package com.atguigu.controller;

import org.apache.commons.io.FileUtils;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.http.HttpHeaders;
import org.springframework.http.HttpStatus;
import org.springframework.http.ResponseEntity;
import org.springframework.stereotype.Controller;
import org.springframework.web.bind.annotation.RequestHeader;
import org.springframework.web.bind.annotation.RequestMapping;
import sun.misc.BASE64Encoder;

import javax.servlet.ServletContext;
import java.io.File;
import java.io.IOException;
import java.io.UnsupportedEncodingException;
import java.net.URLEncoder;

@Controller
@RequestMapping("file")
public class FileController {
    @Autowired
    private ServletContext context;

    /**
     * 处理文件下载的方法
     * ResponseEntity：封装了HTTP的所有响应信息
     * 参数：filename 客户请求的要下载的文件名
     *
     * @RequestHeader("user-agent") String agent 获取的浏览器的请求头
     * 请求头的内容，是浏览器的名字
     */
    @RequestMapping("download")
    public ResponseEntity<byte[]>download(@RequestHeader("user-agent")String agent,String filename) throws IOException {
        //获取要下载文件的绝对路径
        //最大域对象ServletContext方法getRealPath，获取服务器的绝对路径
        String realPath = context.getRealPath("download");
        //File对，将文件的目录和文件名组合为一个对象
        File downloadFile = new File(realPath,filename);
        //创建HTTP响应头对象
        HttpHeaders httpHeaders = new HttpHeaders();
        //设置响应头,通知浏览器，以附件形式处理，不要打开
        httpHeaders.setContentDispositionFormData("attachment",getChineseFileName(filename,agent));
        //设置要下载文件的MIME类型，响应头就是contentType
        //通过文件名，直接获取该文件的MIME类型
        //ServletContext域对象的方法getMimeType
        String mimeType = context.getMimeType(filename);
        httpHeaders.set("contentType",mimeType);
        //创建封装HTTP响应信息的对象
        /**
         * 构造方法传递参数
         * 传递要下载文件的字节数组
         * 传递响应头信息
         * 传递响应状态码
         */

        //Apache的commonsIO工具类的方法，可以将文件，转成数组
        byte[] bytes = FileUtils.readFileToByteArray(downloadFile);
        return new ResponseEntity<byte[]>(bytes,httpHeaders, HttpStatus.OK);
    }


    //将要下载的文件名，处理中文乱码
    private String getChineseFileName(String filename, String agent) throws UnsupportedEncodingException {
        if (agent.contains("MSIE")) {
            // IE浏览器
            filename = URLEncoder.encode(filename, "utf-8");
            filename = filename.replace("+", " ");
        } else if (agent.contains("Firefox")) {
            // 火狐浏览器
            BASE64Encoder base64Encoder = new BASE64Encoder();
            filename = "=?utf-8?B?" + base64Encoder.encode(filename.getBytes("utf-8")) + "?=";
        } else {
            // 其它浏览器
            filename = URLEncoder.encode(filename, "utf-8");
        }
        return filename;
    }
}



